import React from 'react';
import { MenuItem } from '@mui/material';

/**
 * AI模型列表组件
 * 包含所有可用的AI模型选项
 */
export const AIModelMenuItems: React.FC = () => {
  return (
    <>
      <MenuItem value="deepseek-chat">DeepSeek Chat (V3)</MenuItem>
      <MenuItem value="deepseek-reasoner">DeepSeek Reasoner (R1)</MenuItem>
      <MenuItem value="zhipu-glm-4">智谱 GLM-4</MenuItem>
      <MenuItem value="01-yi-chat">零一万物 YI Chat</MenuItem>
      <MenuItem value="qwen-max">通义千问 Qwen2</MenuItem>
      <MenuItem value="moonshot-v1-128k">Moonshot V1-128K</MenuItem>
      <MenuItem value="baichuan-turbo">百川 Turbo</MenuItem>
    </>
  );
};

/**
 * 获取AI模型名称的函数
 * @param modelId 模型ID
 * @returns 模型的显示名称
 */
export const getModelName = (modelId: string): string => {
  const modelNames: Record<string, string> = {
    'deepseek-chat': 'DeepSeek Chat (V3)',
    'deepseek-reasoner': 'DeepSeek Reasoner (R1)',
    'zhipu-glm-4': '智谱 GLM-4',
    '01-yi-chat': '零一万物 YI Chat',
    'qwen-max': '通义千问 Qwen2',
    'moonshot-v1-128k': 'Moonshot V1-128K',
    'baichuan-turbo': '百川 Turbo'
  };
  
  return modelNames[modelId] || modelId;
}; 